from mymath import add


def test_add_1_2():
    r = add(1, 2)
    if r == 3:
        print('good')
    else:
        print('bad')


def test_add_a_b():
    r = add('a', 'b')
    if r == 'ab':
        print('good')
    else:
        print('bad')


if __name__ == '__main__':
    test_add_1_2()
    test_add_a_b()
